package com.cracker.websocket.server.dao;

import com.cracker.websocket.server.entity.GroupEntity;
import com.cracker.websocket.server.entity.UserEntity;
import org.springframework.stereotype.Component;

import java.util.Date;
import java.util.HashSet;
import java.util.Set;

@Component
public class UserDao {
    public UserEntity getByToken(String token) {
        UserEntity userEntity = new UserEntity();
        userEntity.setId((int) (Math.random() * 1000));
        userEntity.setCreatedAt(new Date());
        userEntity.setName(token);
        GroupEntity groupEntity = new GroupEntity();
        groupEntity.setId("2");
        Set<GroupEntity> groupEntities = new HashSet<>();
        groupEntities.add(groupEntity);
        userEntity.setGroupEntity(groupEntities);
        return userEntity;
    }
}
